from LLM_API import LLM_client
from prompt_net import pNet
from img_gen_API import img_model
from vid_gen_API import vid_model
import init_promptes
import json

LLM_API_KEY_main = ""
LLM_API_URL_main = ""
LLM_MODEL_NAME_main = ""

LLM_API_KEY_fast = ""
LLM_API_URL_fast = ""
LLM_MODEL_NAME_fast = ""

IMG_API_KEY = ""
IMG_API_URL = ""
IMG_MODEL_NAME = ""
IMG_SAVE_PATH = ""

VIDEO_API_KEY = ""
VIDEO_API_URL = ""
VIDEO_MODEL_NAME = ""
VIDEO_SAVE_PATH = ""

STORY_PATH = ""

client_main = LLM_client(model_name=LLM_MODEL_NAME_main,api_key=LLM_API_KEY_main,base_url=LLM_API_URL_main)
client_fast = LLM_client(model_name=LLM_MODEL_NAME_fast,api_key=LLM_API_KEY_fast,base_url=LLM_API_URL_fast)
prompts = init_promptes.Prompt()
C2J = init_promptes.C2JSON()

net = pNet(client_main, client_fast, prompts, C2J)

with open(STORY_PATH, 'r', encoding='utf-8') as file:
    story = file.read()
net.get_pardas(story)

with open("data.json", "w") as f:
    json.dump(net.LLM_main_dict, f)

